# define the super class of agent

from config.cfg import Config


class Agent:
    def __init__(self, net_dims: list[int], state_dim: int, action_dim: int, device: str, args: Config):
        self.device = device
        self.state_dim = state_dim
        self.action_dim = action_dim
        self.net_dim = net_dims
        self.learning_rate = args.lr

    def take_action(self, state):
        pass

    def update(self, transition_dict):
        pass

    def soft_update(self, net, target_net):
        pass
